import tools.gligen
import tools.llama3
import tools.op_text
import tools.prompt_i
import tools.question
import tools.select_area
import tools.translate
import tools.common

NODE_CLASS_MAPPINGS = {
    "FCL_PROMPT": tools.llama3.Llama3,
    "FCL_TRANSLATE": tools.translate.Translate,
    "FCL_SELECT_AREA": tools.select_area.SelectArea,
    "FCL_GLIGEN_TEXT_BOX_APPLY": tools.gligen.FclGligenTextBoxApply,
    "FCL_AREA_LIST": tools.common.AreaList,
    "FCL_AREA": tools.common.FclArea,
    "FCL_CONDITIONING_SET_AREA": tools.common.FclConditioningSetArea,
    "FCL_OP_TEXT": tools.op_text.OpText,
    "FCL_QUESTION": tools.question.Question,
    "FCL_PROMPT_I": tools.prompt_i.PromptI
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "FCL_PROMPT": "fcl dev prompt",
    "FCL_TRANSLATE": "fcl dev translate",
    "FCL_SELECT_AREA": "fcl dev select area",
    "FCL_GLIGEN_TEXT_BOX_APPLY": "fcl dev gligen text box apply",
    "FCL_AREA_LIST": "fcl dev area list",
    "FCL_AREA": "fcl dev area",
    "FCL_CONDITIONING_SET_AREA": "fcl dev conditioning set area",
    "FCL_OP_TEXT": "fcl operator text",
    "FCL_QUESTION": "fcl dev question",
    "FCL_PROMPT_I": "fcl dev prompt 1",
}
